from enum import Enum

from git import Repo, RemoteReference


class CheckType(Enum):
    NONE = 0
    CURRENT = 1
    LOCAL = 2
    REMOTE = 11


class CheckResult:
    def __init__(self, repo: Repo, check_type=CheckType.NONE, branch_name: str = None, remote_branch: RemoteReference = None):
        self.repo = repo
        self.check_type = check_type
        self.branch_name = branch_name
        self.remote_branch = remote_branch
